
# class SubTaskConfig():
#     agent_list = [
#
#                      {"team": 0, "tid": 0, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 1, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 2, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 3, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 4, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 5, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 6, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 7, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 8, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 9, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 10, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 11, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 12, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 13, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 14, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 15, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 16, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 17, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 18, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 19, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 20, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 21, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 22, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 23, "type": "Drone", "init_fn_name": "init_air"},
#                      {"team": 0, "tid": 24, "type": "Drone", "init_fn_name": "init_air"},
#
#         {"team": 1, "tid": 0, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 1, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 2, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 3, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 4, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 5, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 6, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 7, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 8, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 9, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 10, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 11, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 12, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 13, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 14, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 15, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 16, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 17, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 18, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 19, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 20, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 21, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 22, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 23, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 24, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 25, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 26, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 27, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 28, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 29, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 30, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 31, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 32, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 33, "type": "Gunner", "init_fn_name": "init_ground"},
#         {"team": 1, "tid": 34, "type": "Gunner", "init_fn_name": "init_ground"},
#     ]
#     obs_vec_length = 23
#     obs_n_entity = 11
#     ActionFormat = 'Multi-Digit'



class SubTaskConfig():
    agent_list = [

       {"team": 0, "tid": 0, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
       {"team": 0, "tid": 1, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
       {"team": 0, "tid": 2, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
       {"team": 0, "tid": 3, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
       {"team": 0, "tid": 4, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},

        {"team": 1, "tid": 0, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 1, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 2, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 3, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 4, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 5, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},
        {"team": 1, "tid": 6, "type": "BasePawn_Air_C", "init_fn_name": "init_ground"},

    ]

    agent_list_all= []
    obs_vec_length = 23
    obs_n_entity = 1
    keyobj_n_entity = 1
    ActionFormat = 'Multi-Digit'
